package com.wulis.common.generator;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.wulis.config.mybatis.plus.MybatisPlusGeneratorConfig;

import lombok.extern.slf4j.Slf4j;

import java.util.HashMap;
import java.util.Map;

/**
 * 代码生成入口
 * 
 * @author WuliBao
 * @date 2020-04-07
 */
@Slf4j
public class MybatisPlusGeneratorEntrance {
    
    public static void main(String[] args) {
        // 模块名称(会在provider下生成包)
        String moduleName = "test.demo";
        // 需要生成的表名及对应的实体类名称<K,V> K:表 V:表名前缀 即生成代码时名称需要忽略的字符
        Map<String, String> tables = new HashMap<>(8);
        tables.put("epc_cost_subject", "epc_");
        tables.put("epc_engineering_project_type", "epc_engineering_");
        // 获取代码生成对象及基础配置(可传模块名，默认取第一个表名的前缀)
        AutoGenerator generator = new MybatisPlusGeneratorConfig().mybatisPlusGenerator(tables, moduleName);
        // 执行代码生成
        generator.execute();
    }
}
